package com.lib.common.utils;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.security.Keys;
import org.springframework.beans.factory.annotation.Autowired;

import java.nio.charset.StandardCharsets;
import java.util.Date;
import java.util.Map;

import javax.crypto.SecretKey;

public class JwtUtils {
    /**
     * 生成 JWT 令牌
     * @param claims 要包含在令牌中的数据
     * @return 生成的 JWT 令牌
     */
    public static String generateToken(String secretKey,long ttlMillis,Map<String, Object> claims) {
        //将编码后的字符串，进行加密处理，成为密钥
        SecretKey SECRET_KEY = Keys.hmacShaKeyFor(secretKey.getBytes(StandardCharsets.UTF_8));
        return Jwts.builder()
                .claims(claims)
                .signWith(SECRET_KEY, SignatureAlgorithm.HS256)
                .expiration(new Date(System.currentTimeMillis() + ttlMillis))
                .compact();
    }

    /**
     * 解析 JWT 令牌
     * @param jwt 要解析的 JWT 令牌
     * @return 解析后的 Claims 对象
     */
    public static Claims parseToken(String secretKey,String jwt) {
        //将编码后的字符串，进行加密处理，成为密钥
        SecretKey SECRET_KEY = Keys.hmacShaKeyFor(secretKey.getBytes(StandardCharsets.UTF_8));
        try {
            return Jwts.parser()
                    .verifyWith(SECRET_KEY)
                    .build()
                    .parseSignedClaims(jwt)
                    .getPayload();
        } catch (Exception e) {
            throw new RuntimeException("Failed to parse JWT: " + e.getMessage());
        }
    }
}